Add (f**king painful) solution to problem 3