Updated gem file to support latest changes