Updates and fixes

Added missing parameters to methods, fixed few bugs and updated docs.