Add support for ax and sreg attributes