I added a test for the median function