Initial implementation of UI for prediction