Small tweak to code and added High Level Design Document