Added documentation, cleaned imports