better with less code :)