tidying up first assignment1