Minor edits plus test 2 cases

Added test cases for test_exercise2.py