I changed inheritance a little bit