[setup.sh] refine my code