withColorChoice now works