Add function to bench