added documentation, finetuning