Updating tests to test more lines of code