[TASK] Cleaned up documentation, fixed some issues