Incorporated pull request comments

Signed-off-by: dpulitano <danielpulitano@gmail.com>