adding new coding for exercise 1