My data structure homework