Re adding weight sharing