Define main correctly