add edit choice functionality