Updated example and moved benchmark time into class