Added initial visualization