Made code more prettier