Simple code cleanup
Use long not int for input