Formatted the code a bit