Added basic visualisation script as a test