Bug no longer an issue + user defined screen size

Didn't change anyting just tried some different image sizes and it now
works. I also added the ability for the user to define the screen size
if it is not defined it is defaulted to 256