added dto and main class