Add mouse input to finish screen

TODO: Add comments