Tidy up
Update docs