Add optional deps
Fix references