Add shop screen and change gameplay