Added documentation and added a separate class.