Updated DAO layer