Add licence header
Updated tests