Added visualisation from base, added functionality.