Added lighting code and added comments