Create separate main class for personal predictions. Also tested on EdLab