Adding duration feature