add assignment 1 basic code