improved state visualisation