assignment 2 work. Almost completed. Fixing bugs.