Updated some code to add rendering of levels