add assignment 1 base code