Fix functionality for new-game button