Updated docs
Tidy up