Added solution to problem 1.

Signed-off-by: Andrew Coomes <ajc@ajcdrift.com>