Added restart scene + some graphic additions