solved problem four

Signed-off-by: Arvind Chembarpu <achembarpu@gmail.com>