Finished dropout implementation