adding code, fixing header