Fixed problems not working

Fixes #63 - some bugfixes with how problems was implemented