Using simple 'Animate Item' and 'Show/Hide' interactions to have the boxes disappear and realign accordingly. Can be used as a base on how you can build notifications or lists to realign themselves.

to demonstrate the navigation between the screen and state transitions

A simple spaceship animation using 'Animate Item' & states. Design Source: https://www.sketchappsources.com/free-source/3673-rocket-illustration-sketch-freebie-resource.html