Update Rakefile

Update rakefile as well.