Added conclusion and made code consistent