Tidy up v2.0