This program can further be improved by off-loading some more operations from the CPU to the GPU. But this will require more time and research.