Add note on test to README
Tidy up comments