Update comment
Tidy up