Add calulating entropy in some other class.