Applied beter formatting to updated code.