Fixed remaining column issues
Merge pull request #14 from bbai/AddColumns

Fixed remaining column issues