Cleaned up your code.