Adding assingment forcomp, not working