fixed converting it to a gem, still need to make sure it works properly