Added benchmark code