Made code reload when level restarts