Adding tests that pass