Did add a main loop so it work more like a game now