Added test for create prediction