Trying to make the dropdown more flexible