Replaced doneButton with doneButton2 for testing purposes