Added better functionality in levels