add some of the code