Updated main for clairty