I got the code to work properly : )