fix headers as per remarks in #117

Signed-off-by: Fotis Georgatos <fotis.georgatos@uni.lu>