Updated docs + fixed stupid refactoring issue