Checking in W4 - Assignment 3 solution