WIP
Fix warnings
Update docs