Adding tests to problem 11