WIP Add rake tasks