Add support for rake test