Adding documentation
Cleaned up