updated readMe and tidied up code