Cleaned up the code a bit; removed unused image