update naming of main arguments
add code for testing
add code for testing
update code to include Rspec testing