Bug fixes

Updating runner code using wine dataset as a test