Highlighted source code

I highlighted the source codes for better understanding.