Add latex output
Update assignment