Code cleanup und corrected indentation