Další úprava dialogu, základ již funguje