Add account type to account model