Added update function for game