Small changes to modal