Need to convert this to use a modal