Change 'return' to 'print' from main