Adding some data structure code