Adding tests to test suit