Fix main-method