JDialog fixes, some bugs