added class comment and made the code look a lite better
added class comment