Made the code a little bit more convenient

also using the duration as intended