Added test in draft script