A. W. Roberts, Union officer