Re-formatted the code