Cleaned and added documentation