simple code for interview