Implemented test correctly,

+ corrected indentation