Adding assignment one code