Importing HW2 codebase