reverted code. added another feature visualisation