Fix linting errors
Fix output