Updated rakefile to include tests