Fix variable names
Update docs