Cleaned Code
Minor Cleaning