fixed compilation errors in both files