Homework for data structures class