Update rakefile for testing