Add benchmark
Minor fixes