Update bitdepth warning
Fix PEP8 and unused imports