Update licence header
Minor clean up