Added functionality to level complete screen