Added code for actual test