Added predict button
Added button press
Added button action