More changes after client review

Signed-off-by: Maikol Guzman <mikeguzman@gmail.com>