Done button changes states properly