fixed some problems with the final level