added more tests before submission