Added ball prediction