Added initial Lab 4 code