Tidied up code + updated readme