Final wrap up fixes