Added round diffuser; removed copy pasted main method