working on prob class
fix docs