Separate visualise code into its own module and function