Seems  a much better way and also avoid hardcode.