updated the code that we worked on sunday