Update code.c fixed #2

Reopened and fixed #2 issue