Added cell edit and visualization