Update for verson 0.3.1