Adding the wait() method \o/!