I took this code as it was, but sure... can fix it up.