Solved problem one, waiting for the objective of problem 2