Added second problem... currently working