updated with small fixed for code section