Updated Unit/Acceptance Test