update setup script for new code structure