adding base code that we need to add in lab4