Solution works for examples now

Not for whole challenge though