Add some code to work modal