Checking in the code I have so far