Done adding gym stuff, now make it work