Fix type signatures
A little bit of docs