Refactored main function.
Return keyword.