changed the input to match julia