Finished version2

Added visualiser