Work on code to check the answer