Changed Training to be cleaner, still not working