started adding code to problem