Rewrite axis using d3 api