Remove wait time from main.cpp