Only training classes distribution for now