Corrections to get code running