Use mixin instead of whole code