Fix issue # 10

Still has issues if '<code>' has any additional spaces or attributes, but works now.