Added functionality to later add a pause menu