Fixed alignment of code and added comments