improved visualisation, fixed mistake