Fix logging
Develop prediction generation logic