Small fix for prediction pop-up. Fixed and closed #27