quick code - not anything great - View it on GitHub