Still not perfect yet but it works.