clean up
improved spec to include test for arguments passed in