Add support for rake task