Documentation
(Slightly) better comments