Flat-top pole write head design