Added license headers and removed unused imports