problem 2 solved, updated readme