Cleaned up how levels are reset after they are completed