Cleaned & added documentation