Added code to hide keyboard on done