Hopefully finished code
Added screenshots
Finished code