Getting model diagrams and added switch block.