added __all__ tag
Added documentation
Created a problem class for Wright FIsher model