Fixing buttons in win and lose screen