add some tests
add fast prediction