Data Structure class homework