Updated the readme a bit
Added comment to main class