Simplfied the exercise