Adding benchmark code