Formatted code a bit better.