My edited first assignment