Clean up code Add proper documentation
Clean up code
Add proper documentation