[FEATURE] Initial commit with working models