making edits after the test cases came out