Add utils.py with wait_for_port function