Add Homework class and main test file