Make it just component to wrap and simulate devices based on a given size.
Leave the UI part as either a separate component (which could be in the same package) or remove it entirely.
This will allow integration on different projects more easily.
Makes sense?