Added benchmarking for prob1.
Put import in #[cfg(test)] block.
Added benchmarking and refactored.