Cleaned up benchmark