Cleanup the code
Removed unnecessary gems