Added a bunch of references
Added some checks for the dropout param