- Fixed a few issues with displaying the 2D simulation