Fixing STI functionality
Updated Schema to include functional STI