Move the rake task into lib