code got more better :)