fixed main. #5