Improved: Class can be specified in megatron.sh using --class.