Cleaned up code. Whitespaces etc.