Use the structure of GYM and try refactor the model in this. This will make it easier for people to use the package.