Tidied up solution and added more notes to readme