ER diagram small update