Reference to the paper + small fixes