Update rakefile to work