Homework for data structure course