Counting - Fix main function location