add code to main gem file