added new features

like probabilty