Add tests for medium