Added package info and class diagram