added mean calculation
added predictions method