Updated test task in Rakefile