Added another piece of code and two tests. Unable to test because my ruby command line was not working