Added code structure for levels