added a custom prediction button