Added parameters for visualiser to know what size should it be.
Removed main method in Visualiser.