updated project to work as a gem