Add tests and controller updates for editing users