Code to test predictions