Added just wait function