Code ready to be merged with don's. Working screenshot