Endring og utvidelse av main

.