trying to fix done button