Add license and cleaned class diagram