Modified after_script