improved the ER diagram