Moved s-box to constant device memory,  cleaned up comments, other things