Made it so that it works as a gem