Added
+ fiksed timing on end
+ fixed so it goes back to menu after gameover