Updated wrapper script