Remove useless argument
Add solution for test problem