Added a test for median