Still debugging dropout - editing locally because for some reason it works for me. Pardon the extra commits.