Added @tomMalbran's code. Looks nice now.