Updated account update functionality and added supporting tests