Added a restart button and changed gravity