Add missing license header
Minor clean up