Add TODOs
Add inheritance