writing a few more tests before submission