Style code and add notes