Moving puzzle ui container out of game template