Fix/header

done for #15 .