Removed actual plotting part from tests