Added problem two & three, plus tests. Also added gen-idea plugin and removed redundant Main.scala