add problem tests. fixed #2
change problem model