updating vis code again to make look nicer