Improvind the code. =)