Added code for modal