Added class for level loading