Lock works, first two test case done for part 1