Updated tests for median