Added modal functionality