Modified the Test Controller