Add missing license headers + PEP8 formatting