going on checking in code from last assignment