done minor changes - suggested by sujit