Add header & tests