Make code directive working