Cleaner way of getting the 'timeout' argument