Added visualization loop, F2 to exit