fixes #4 & ERD