trying to build toy model for testing