Fix indentation error & make it look a little nicer